package com.ruoyi.common.utils;

import com.ruoyi.common.utils.spring.SpringUtils;
import com.ruoyi.common.websocket.MessageEndpoint;
import com.ruoyi.common.websocket.SocketMessageVo;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

/**
 * @author ab785
 * createTime: 2022-1-25.
 * description: socket工具类
 */
@Component
public class WebsocketUtils {

    private static MessageEndpoint messageEndpoint;

    @PostConstruct
    public void init() {
        WebsocketUtils.messageEndpoint = SpringUtils.getBean(MessageEndpoint.class);
    }

    /**
     * 通知前端刷新消息盒子
     */
    public static void notifyToReloadMessageDrawer() {
        SocketMessageVo<String> messageVo = new SocketMessageVo<>("reloadMessageDrawer", null);
        messageEndpoint.sendMessageToAllUsers(messageVo);
    }

}
